Geometric Shapes in Python